Small fix for drawing