Minor test adjustment