Update to fix caption