Fix stop button not appearing