diff --git a/hpvm/scripts/hpvm_installer.py b/hpvm/scripts/hpvm_installer.py
index bf879cdb1db3151d94acfd4f53b36aa6ea6ef025..11ad3045528543d8398828394cc7236fc849bb30 100755
--- a/hpvm/scripts/hpvm_installer.py
+++ b/hpvm/scripts/hpvm_installer.py
@@ -74,6 +74,7 @@ def prompt_args():
     def parse_int(s: str):
         try:
             v = int(s)
+            return v
         except ValueError:
             return None
         if v <= 0:
diff --git a/hpvm/test/unitTests/TwoLevel.ll b/hpvm/test/unitTests/TwoLevel.ll
index 0289319517b7d6a1f83f7b64d615bcbd72630821..840a2b5685d33d02584b72d96482fedda9a52fb6 100644
--- a/hpvm/test/unitTests/TwoLevel.ll
+++ b/hpvm/test/unitTests/TwoLevel.ll
@@ -10,7 +10,7 @@ target triple = "x86_64-unknown-linux-gnu"
 define dso_local void @Func1(i32* %In, i64 %Insize, i32* %Out, i64 %Outsize) #0 {
 entry:
   tail call void @__hpvm__hint(i32 1) #3
-  tail call void (i32, ...) @__hpvm__attributes(i32 1, i32* %In, i32* %Out, i32 1, i32* %Out) #3
+  tail call void (i32, ...) @__hpvm__attributes(i32 2, i32* %In, i32* %Out, i32 1, i32* %Out) #3
   %0 = load i32, i32* %In, align 4, !tbaa !2
   store i32 %0, i32* %Out, align 4, !tbaa !2
   tail call void (i32, ...) @__hpvm__return(i32 1, i32* %Out) #3