Fix unit tests

[planetA]

As of now units tests compile, but file. The tests should be fixed or removed. Better fix.