fix Class usage