Fix caption change height functionality