improve #transformDoit

pavel-krivanek · Nov 8, 2019 · eebae22 · eebae22
1 parent b579d0f
commit eebae22
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/OpalCompiler-Core/OpalCompiler.class.st b/src/OpalCompiler-Core/OpalCompiler.class.st
@@ -737,12 +737,11 @@ OpalCompiler >> source: aString [
 { #category : #'public access' }
 OpalCompiler >> transformDoit [
 	"if we compile an expression, we wrap it in a method here"
-	self compilationContext noPattern ifFalse: [ ^self ].
+	self compilationContext noPattern ifFalse: [ ^ast ].
 	ast := context 
 		ifNil: [ast asDoit] 
 		ifNotNil: [ast asDoitForContext: context].
-	ast compilationContext: self compilationContext.
-	^ast
+	^ast compilationContext: self compilationContext
 ]
 
 { #category : #'old - deprecated' }