diff --git a/test/api/Terminal.api.ts b/test/api/Terminal.api.ts
index b2157b1e..29a361a9 100644
--- a/test/api/Terminal.api.ts
+++ b/test/api/Terminal.api.ts
@@ -83,11 +83,9 @@ describe('API Integration Tests', function(): void {
       window.term.onData(e => calls.push(e));
       window.term.paste('foo');
       window.term.paste('\\r\\nfoo\\nbar\\r');
-      window.term.write('\\x1b[?2004h');
-      // TODO: Use promise/callback for write when we support that
-      // Force sync write
-      window.term._core._innerWrite();
-      window.term.paste('foo');
+      window.term.write('\\x1b[?2004h', () => {
+        window.term.paste('foo');
+      });
     `);
     assert.deepEqual(await page.evaluate(`window.calls`), ['foo', '\rfoo\rbar\r', '\x1b[200~foo\x1b[201~']);
   });