Fix draw shape bug

Cannot draw shape.