Forgot to rename the actual class