diff --git a/src/build.py b/src/build.py
index b9b224f157..cb8413c468 100755
--- a/src/build.py
+++ b/src/build.py
@@ -252,7 +252,7 @@ def make_package(args):
 
     # Figure out if application has service part
     service = False
-    directory = args.dir
+    directory = args.dir if public_version else args.private
     if directory:
         service_main = join(realpath(directory), 'service', 'main.py')
         if os.path.exists(service_main) or os.path.exists(service_main + 'o'):