use correct function in unit test