fix compilation

The unit-tests  still pass, however they are probably not checking enough.