Fix play button bug