Updating tests to reflect hook change