Fix buttons not working