diff --git a/lib/plugins/aws/package/lib/saveCompiledTemplate.test.js b/lib/plugins/aws/package/lib/saveCompiledTemplate.test.js
index e66bb8b4a..ce039f492 100644
--- a/lib/plugins/aws/package/lib/saveCompiledTemplate.test.js
+++ b/lib/plugins/aws/package/lib/saveCompiledTemplate.test.js
@@ -33,6 +33,7 @@ describe('#saveCompiledTemplate()', () => {
   });
 
   afterEach(() => {
+    saveCompiledTemplate.provider.naming.getCompiledTemplateFileName.restore();
     saveCompiledTemplate.serverless.utils.writeFileSync.restore();
   });
 
@@ -45,7 +46,7 @@ describe('#saveCompiledTemplate()', () => {
 
     return saveCompiledTemplate.saveCompiledTemplate().then(() => {
       expect(getCompiledTemplateFileNameStub.calledOnce).to.equal(true);
-      expect(writeFileSyncStub.calledWithExactly(filePath, 'compiled content'));
+      expect(writeFileSyncStub.calledWithExactly(filePath, 'compiled content')).to.equal(true);
     });
   });
 });