Fix the class transformer