diff --git a/src/Behat/Behat/Definition/Annotation/Transformation.php b/src/Behat/Behat/Definition/Annotation/Transformation.php
index dd481eb75..92d0322dd 100644
--- a/src/Behat/Behat/Definition/Annotation/Transformation.php
+++ b/src/Behat/Behat/Definition/Annotation/Transformation.php
@@ -3,6 +3,7 @@
 namespace Behat\Behat\Definition\Annotation;
 
 use Behat\Gherkin\Node\TableNode;
+use Behat\Gherkin\Node\PyStringNode;
 
 use Behat\Behat\Definition\TransformationInterface,
     Behat\Behat\Context\ContextInterface,