Fix style issues in unit model.