fixing failed tests