diff --git a/lib/generate.js b/lib/generate.js
index 72f0939..2a9c398 100644
--- a/lib/generate.js
+++ b/lib/generate.js
@@ -59,7 +59,7 @@ module.exports = async (root, pkg, release, customName) => {
 
   const serviceFile = generateServiceFile(root, customPackage);
   const specFile = generateSpecFile(specsDirectory, customPackage, release);
-  const archiveWhitelist = getArchiveWhitelist(pkg);
+  const archiveWhitelist = getArchiveWhitelist(customPackage);
 
   await archiver.compress(root, sourcesArchive, archiveWhitelist);
 
diff --git a/test/generate.js b/test/generate.js
index e47fc92..97690f3 100644
--- a/test/generate.js
+++ b/test/generate.js
@@ -112,7 +112,7 @@ describe('generate', () => {
       {
         files: undefined,
         main: 'index.js',
-        service: 'my-cool-api.service'
+        service: 'penguin.service'
       }
     );
   });
@@ -135,6 +135,24 @@ describe('generate', () => {
     );
   });
 
+  it('passes files and main values from the package.json to archiver using custom name', async () => {
+    await generate('/path/to/project', pkgWithWhitelist, null, 'penguin');
+    sandbox.assert.calledWith(
+      archiver.compress,
+      '/path/to/project',
+      '/path/to/project/SOURCES/penguin.tar.gz',
+      {
+        main: 'server.js',
+        files: [
+          'lib',
+          'routes',
+          'index.js'
+        ],
+        service: 'penguin.service'
+      }
+    );
+  });
+
   it('creates the service file with a custom name if specified', async () => {
     await generate('/path/to/project', pkg, 1, 'penguin');
     sandbox.assert.calledWith(