fixed tests! hooray!