fix for test failures