trying to fix failed tests