Adjust tests for 204