fix reference in unit test