Use setup method in tests now