Fixing caption problem