Made test set up more concise.