Bug fix for model test script