Minor test setup change