Fix patch for all models. Previous one caused unusual behavior.