Just 2 tiny changes