fix unit test reference error