Added missing play state check