Fix oops that prevents display restoration.