Minor change to pass test