diff --git a/src/DoctrineContext.php b/src/DoctrineContext.php
index 2d17f90..2fdc15a 100644
--- a/src/DoctrineContext.php
+++ b/src/DoctrineContext.php
@@ -10,7 +10,7 @@
  *
  * @see http://behat.org/en/latest/quick_start.html
  */
-class FeatureContext extends BaseContext
+class DoctrineContext extends BaseContext
 {
     static private $max_queries;
     static private $query_count = 0;