Fixed test that failed due to a change in the framework