Write test in a slightly different way