fixed tests to be all green