Forgot to remove one class