fix unit test (much better solution...)