Move test setup into tests themselves