Press the button, then wait for the light. That's correct.

that 是指示代词，指代前面提到的整个操作过程（按下按钮并等待灯亮）。 correct 是形容词，意思是"正确的、对的"。 that's correct 意思是"那样是正确的"，用来确认前面描述的操作步骤是正确的。