Bug fix for model verification script