Fix buttons disabling stuff