Fixed tests for revised API call.