#11 fix graphical bug