fixed tests

API updates oops