fixed compilation problem in unit test