diff --git a/src/demo.py b/src/demo.py
index 57567d2..7fc6da4 100644
--- a/src/demo.py
+++ b/src/demo.py
@@ -1,14 +1,14 @@
 from tpre import *
 
 # 1
-pk_a, sk_a = GenerateKeyPair(1, ())
+pk_a, sk_a = GenerateKeyPair()
 m = b"hello world"
 
 # 2
 capsule_ct = Encrypt(pk_a, m)
 
 # 3
-pk_b, sk_b = GenerateKeyPair(1, ())
+pk_b, sk_b = GenerateKeyPair()
 
 N = 5
 T = 2