diff --git a/capsulflask/__init__.py b/capsulflask/__init__.py
index 6b737db..3d84f38 100644
--- a/capsulflask/__init__.py
+++ b/capsulflask/__init__.py
@@ -68,7 +68,12 @@ stripe.api_key = app.config['STRIPE_SECRET_KEY']
 stripe.api_version = app.config['STRIPE_API_VERSION']
 
 app.config['FLASK_MAIL_INSTANCE'] = Mail(app)
-app.config['VIRTUALIZATION_MODEL'] = virt_model.MockVirtualization()
+
+if app.config['VIRTUALIZATION_MODEL'] == "shell_scripts":
+    app.config['VIRTUALIZATION_MODEL'] = virt_model.ShellScriptVirtualization()
+else:
+    app.config['VIRTUALIZATION_MODEL'] = virt_model.MockVirtualization()
+
 app.config['BTCPAY_CLIENT'] = btcpay.Client(api_uri=app.config['BTCPAY_URL'], pem=app.config['BTCPAY_PRIVATE_KEY'])
 
 from capsulflask import db