TypeScriptToLua
diff --git a/‎dist/Transpiler.js‎
Lines changed: 6 additions & 0 deletions b/‎dist/Transpiler.js‎
Lines changed: 6 additions & 0 deletions
diff --git a/‎src/Transpiler.ts‎
Lines changed: 7 additions & 0 deletions b/‎src/Transpiler.ts‎
Lines changed: 7 additions & 0 deletions
@@ -784,6 +784,12 @@ var LuaTranspiler = /** @class */ (function () {
         node.members.filter(ts.isMethodDeclaration).forEach(function (method) {
             result += _this.transpileMethodDeclaration(method, className + ".");
         });
+        // Check if the class should be returned
+        var isExport = node.modifiers && node.modifiers.some(function (_) { return _.kind == ts.SyntaxKind.ExportKeyword; });
+        var isDefault = node.modifiers && node.modifiers.some(function (_) { return _.kind == ts.SyntaxKind.DefaultKeyword; });
+        if (isExport && isDefault) {
+            result += this.indent + ("return " + className + "\n");
+        }
         return result;
     };
     LuaTranspiler.prototype.transpileConstructor = function (node, className, instanceFields) {
 
@@ -852,6 +852,13 @@ export class LuaTranspiler {
             result += this.transpileMethodDeclaration(method, `${className}.`);
         });
 
+        // Check if the class should be returned
+        const isExport = node.modifiers && node.modifiers.some(_ => _.kind == ts.SyntaxKind.ExportKeyword);
+        const isDefault = node.modifiers && node.modifiers.some(_ => _.kind == ts.SyntaxKind.DefaultKeyword);
+        if (isExport && isDefault) {
+            result += this.indent + `return ${className}\n`;
+        }
+
         return result;
     }