Fix failed tests

ok got it