fix  tests