fixed tests they all pass locally