Fix - Added test for button press.