another fix , only 2 test not working