Tiny bit of tidying