Maybe factor out in setUp