Fixed restore_hazards_test.py

Fixed a test broken by the change in api. See