slight fixed to test suite