adjust tests for API