--- a/toolkit/components/payments/test/browser/head.js
+++ b/toolkit/components/payments/test/browser/head.js
@@ -157,15 +157,14 @@ async function spawnInDialogForMerchantT
     ok(!!request.requestId, "Got a payment request with an ID");
 
     await spawnTaskInNewDialog(request.requestId, dialogTaskFn, taskArgs);
   });
 }
 
 add_task(async function setup_head() {
   await profileStorage.initialize();
-
-  SimpleTest.registerCleanupFunction(function cleanup() {
+  registerCleanupFunction(function cleanup() {
     paymentSrv.cleanup();
     profileStorage.addresses._nukeAllRecords();
     profileStorage.creditCards._nukeAllRecords();
   });
 });