Made test_model.py restartable.