Minor change to tests