Fixing the graphics