Minor test correction for last commit