Fixes test unit test