Fix unit rendering