Fixes test unit tests
Fixes test unit tests