diff --git a/lib/contracts/code_templates/main-context.js.ejs b/lib/contracts/code_templates/main-context.js.ejs
index 557e62b65..53fa09d22 100644
--- a/lib/contracts/code_templates/main-context.js.ejs
+++ b/lib/contracts/code_templates/main-context.js.ejs
@@ -1 +1 @@
-var __mainContext = __mainContext || this;
+var __mainContext = __mainContext || (typeof self !== 'undefined' ? self : this);