Fix bug with captions