Unit tests fix

Replaced two assertTrue with assertEquals, to prevent a compilation
error.