Fix button disabling