fixed tests

thank you!!