Fix draw function.