package geb.transform.implicitassertions;

import geb.transform.implicitassertions.ImplicitAssertionsTransformationVisitor;
import org.codehaus.groovy.ast.ClassNode;
import org.codehaus.groovy.transform.GroovyASTTransformation;
import org.codehaus.groovy.control.CompilePhase;
import org.codehaus.groovy.transform.ASTTransformation;
import org.codehaus.groovy.ast.ASTNode;
import org.codehaus.groovy.control.SourceUnit;

@GroovyASTTransformation(phase = CompilePhase.CANONICALIZATION)
public class ImplicitAssertionsTransformation implements ASTTransformation {
    public void visit(ASTNode[] astNodes, SourceUnit sourceUnit) {
		for (ClassNode classNode : sourceUnit.getAST().getClasses()) {
			classNode.visit(new ImplicitAssertionsTransformationVisitor(sourceUnit));
		}
    }
}
@@ -40,89 +40,8 @@
 s;%0A%0A
-import geb.transform.implicitassertions.ImplicitAssertionsTransformationVisitor;%0A
 impo
@@ -67,21 +67,19 @@
 ovy.ast.
-Class
+AST
 Node;%0Aim
@@ -107,41 +107,21 @@
 ovy.
-transform.GroovyASTTransformation
+ast.ClassNode
 ;%0Aim
@@ -167,16 +167,63 @@
 ePhase;%0A
+import org.codehaus.groovy.control.SourceUnit;%0A
 import o
@@ -301,66 +301,41 @@
 ovy.
-ast.ASTNode;%0Aimport org.codehaus.groovy.control.SourceUnit
+transform.GroovyASTTransformation
 ;%0A%0A@
@@ -624,16 +624,24 @@
 de.visit
+Contents
 (new Imp

Use correct visit method