Fixed issue of units not being drawn