Fix half of the pause / stop button issue