Fixing unit rendering.