fixed tests :D