diff --git a/lib/plugins/aws/deployFunction/tests/index.js b/lib/plugins/aws/deployFunction/tests/index.js
index 97be56524..08bafcaf0 100644
--- a/lib/plugins/aws/deployFunction/tests/index.js
+++ b/lib/plugins/aws/deployFunction/tests/index.js
@@ -122,12 +122,12 @@ describe('AwsDeployFunction', () => {
 
       awsDeployFunction.pkg = pkg;
 
-      const packageServiceStub = sinon
-        .stub(pkg, 'packageService').returns(BbPromise.resolve());
+      const packageFunctionStub = sinon
+        .stub(pkg, 'packageFunction').returns(BbPromise.resolve());
 
       return awsDeployFunction.zipFunction().then(() => {
-        expect(packageServiceStub.calledOnce).to.be.equal(true);
-        awsDeployFunction.pkg.packageService.restore();
+        expect(packageFunctionStub.calledOnce).to.be.equal(true);
+        awsDeployFunction.pkg.packageFunction.restore();
       });
     });
   });