diff --git a/leo/package/src/source/main.rs b/leo/package/src/source/main.rs
index 88549aebe4..37fa50c389 100644
--- a/leo/package/src/source/main.rs
+++ b/leo/package/src/source/main.rs
@@ -66,15 +66,18 @@ impl MainFile {
             .map_err(PackageError::io_error_main_file)?)
     }
 
+    // TODO: Generalize to other networks.
     fn template(&self) -> String {
         format!(
             r#"// The '{}' main function.
-transition main(public a: u32, b: u32) -> u32 {{
-    let c: u32 = a + b;
-    return c;
+program {}.aleo {{
+    transition main(public a: u32, b: u32) -> u32 {{
+        let c: u32 = a + b;
+        return c;
+    }}
 }}
 "#,
-            self.package_name
+            self.package_name, self.package_name
         )
     }
 }