Minor test correction