fixing draw tests