diff --git a/modules/gui/pymod/traj.py b/modules/gui/pymod/traj.py
index 6df252b35e7db3501fd374b74cbb765d5049b28b..1fcb1c77959528cdda00f05140f89902d4c86863 100644
--- a/modules/gui/pymod/traj.py
+++ b/modules/gui/pymod/traj.py
@@ -21,12 +21,13 @@ from PyQt4.QtGui import *
 from ost import *
 
 class TrajWidget(QWidget):
-  def __init__(self, traj=None, render_mode=gfx.SIMPLE, parent=None):
+  def __init__(self, traj=None, render_mode=gfx.SIMPLE, sel='', parent=None):
     QWidget.__init__(self, parent, Qt.Tool)
     self.render_mode=render_mode
     vb=QVBoxLayout()
     hb=QHBoxLayout()
     hb2=QHBoxLayout() 
+    self.selection=sel
     self.callback=None   
     self._slider=QSlider(self)
     self._slider.setOrientation(Qt.Horizontal)
@@ -116,7 +117,8 @@ class TrajWidget(QWidget):
     self._traj=traj
     if self._traj:
       ev=traj.GetEntity()
-      self.gfx_entity=gfx.Entity("mol",self.render_mode, ev)
+      self.gfx_entity=gfx.Entity("mol",self.render_mode,
+                                 ev.Select(self.selection))
       gfx.Scene().Add(self.gfx_entity)
       gfx.Scene().CenterOn(self.gfx_entity)
       # enable the blur effect