fix classes name