fixed bug that wa causing test failures