Updated tests to pass with the new API