Minor test fix