Added the flag so button works now