diff --git a/lib/plugins/install/install.test.js b/lib/plugins/install/install.test.js
index 94193188c..a642bb063 100644
--- a/lib/plugins/install/install.test.js
+++ b/lib/plugins/install/install.test.js
@@ -42,7 +42,7 @@ describe('Install', () => {
       });
     });
 
-    it('should set new service in serverless.yml and name in package.json', (done) => {
+    it('should set new service in serverless.yml and name in package.json', () => {
       const defaultServiceYml =
         'service: service-name\n\nprovider:\n  name: aws\n';
       const newServiceYml =
@@ -58,18 +58,16 @@ describe('Install', () => {
       serverless.utils.writeFileSync(packageFile, { name: defaultServiceName });
       fs.writeFileSync(serviceFile, defaultServiceYml);
 
-      install.renameService(newServiceName, servicePath)
+      return install.renameService(newServiceName, servicePath)
         .then(() => {
           const serviceYml = fs.readFileSync(serviceFile, 'utf-8');
           const packageJson = serverless.utils.readFileSync(packageFile);
           expect(serviceYml).to.equal(newServiceYml);
           expect(packageJson.name).to.equal(newServiceName);
-          done();
-        })
-        .catch(done);
+        });
     });
 
-    it('should set new service in commented serverless.yml and name in package.json', (done) => {
+    it('should set new service in commented serverless.yml and name in package.json', () => {
       const defaultServiceYml =
         '# comment\nservice: service-name #comment\n\nprovider:\n  name: aws\n# comment';
       const newServiceYml =
@@ -85,15 +83,13 @@ describe('Install', () => {
       serverless.utils.writeFileSync(packageFile, { name: defaultServiceName });
       fs.writeFileSync(serviceFile, defaultServiceYml);
 
-      install.renameService(newServiceName, servicePath)
+      return install.renameService(newServiceName, servicePath)
         .then(() => {
           const serviceYml = fs.readFileSync(serviceFile, 'utf-8');
           const packageJson = serverless.utils.readFileSync(packageFile);
           expect(serviceYml).to.equal(newServiceYml);
           expect(packageJson.name).to.equal(newServiceName);
-          done();
-        })
-        .catch(done);
+        });
     });
   });