Added back a function accidentally removed from the test file