Made using tests a bit more DRY