fixed compilation error in unit test