tests fixed

tests fixed