fixes test function