Update to match updated template