Fix caption generation code