fix for unit test