fixed tests whoops