fix for failed tests