forgot to fix unit test due to previous change