Slight bit of tidying