But call the correct function in the test