Correction to piece drawing logic