Fix stuck buttons bug