Added callbacks to test contexts (HAZ)