Fix: defineClass method returnType