Fix output colour restore