Fixes test suite