Replay button fix