More fixes for tests..
Fixed compilation error