fix failing wit unittests