Forgot to change the class name