Forgot to rename the class