Fix freeze when playing