[minor] wnf refactor, fixes in demo
diff --git a/utilities/demo/make_erpnext_demo.py b/utilities/demo/make_erpnext_demo.py
index 4a72e4b..50c66a3 100644
--- a/utilities/demo/make_erpnext_demo.py
+++ b/utilities/demo/make_erpnext_demo.py
@@ -7,7 +7,9 @@
 
 def make_demo_app(site=None):
 	webnotes.mute_emails = 1
-	webnotes.connect(site)
+	webnotes.init(site=site)
+	webnotes.connect(db_name=webnotes.conf.demo_db_name, site=site)
+
 	utilities.demo.make_demo.make(reset=True, simulate=False)
 	# setup demo user etc so that the site it up faster, while the data loads
 	make_demo_user()
@@ -116,4 +118,6 @@
 	webnotes.conn.commit()
 
 if __name__=="__main__":
-	make_demo_app()
+	import sys
+	site = sys.argv[1:]
+	make_demo_app(site=site and site[0] or None)