Restored not training mode