Fixing Issue 284

This commit fixes #284
itemisCREATE · Aug 23, 2019 · 5270f46 · 5270f46
1 parent 415a36a
commit 5270f46
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/...m.yakindu.solidity.ui/src/com/yakindu/solidity/ui/quickfix/SolidityQuickfixProvider.xtend b/...m.yakindu.solidity.ui/src/com/yakindu/solidity/ui/quickfix/SolidityQuickfixProvider.xtend
@@ -469,10 +469,12 @@ class SolidityQuickfixProvider extends ExpressionsQuickfixProvider {
 		acceptor.accept(issue, 'Add default solidity pragma', 'Add solidity pragma ' + solcVersion + '.', null,
 			new ISemanticModification() {
 				override apply(EObject element, IModificationContext context) throws Exception {
-					if (element instanceof SourceUnit) {
-						element.fixPragmaSolidityDirective
-					} else if (element.eContainer instanceof SourceUnit) {
-						(element.eContainer as SourceUnit).fixPragmaSolidityDirective
+					if (element.eContainer instanceof SourceUnit) {
+						val sourceUnit = element.eContainer as SourceUnit
+						val pragma = createPragmaSolidityDirective => [
+						version = solcVersion
+						]
+						sourceUnit.pragma += pragma
 					}
 				}
 			})