fixed tests :)