Forgot to rename the class when changing file names