Adding test for set_model_state