fix for local model bug