Forgot to change string compare