Button test fix