Small fix to draw condition