Use correct thing for test setup