Fix API test.
Fix API tests to pass
Add comment