Forgot to alter the class name, whoops