Fixing regression on models