Trivial bug for restore cd

Hi:
restore cd 2.1.1 I saw changelog “Show error message if disk is too small during restore”.

it is working for mbr restore, but under gpt restore it didn’t show message about size, it just shows error
“Error writing GPT”.