diff --git a/src/DMS/Bundle/FilterBundle/Loader/ContainerAwareLoader.php b/src/DMS/Bundle/FilterBundle/Loader/ContainerAwareLoader.php
index d0709d9..49a6f5d 100644
--- a/src/DMS/Bundle/FilterBundle/Loader/ContainerAwareLoader.php
+++ b/src/DMS/Bundle/FilterBundle/Loader/ContainerAwareLoader.php
@@ -27,7 +27,7 @@ public function setContainer(?ContainerInterface $container = null): void
      *
      * @throws UnexpectedValueException
      */
-    public function getFilterForRule(Rule $rule)
+    public function getFilterForRule(Rule $rule): BaseFilter
     {
         $filterIdentifier = $rule->getFilter();
 
diff --git a/src/DMS/Bundle/FilterBundle/Tests/Loader/ContainerAwareLoaderTest.php b/src/DMS/Bundle/FilterBundle/Tests/Loader/ContainerAwareLoaderTest.php
index 2f32976..5bf3dab 100644
--- a/src/DMS/Bundle/FilterBundle/Tests/Loader/ContainerAwareLoaderTest.php
+++ b/src/DMS/Bundle/FilterBundle/Tests/Loader/ContainerAwareLoaderTest.php
@@ -8,7 +8,6 @@
 use DMS\Filter\Rules\StripTags;
 use PHPUnit\Framework\MockObject\MockObject;
 use PHPUnit\Framework\TestCase;
-use stdClass;
 use Symfony\Component\DependencyInjection\ContainerInterface;
 
 class ContainerAwareLoaderTest extends TestCase
@@ -20,11 +19,11 @@ class ContainerAwareLoaderTest extends TestCase
     public function testGetFilterForRule(): void
     {
         $this->container->expects($this->once())->method('has')->will($this->returnValue(true));
-        $this->container->expects($this->once())->method('get')->will($this->returnValue(new stdClass()));
+        $this->container->expects($this->once())->method('get')->will($this->returnValue(new StripTagsFilter()));
 
         $filter = $this->loader->getFilterForRule(new StripTags());
 
-        $this->assertInstanceOf(stdClass::class, $filter);
+        $this->assertInstanceOf(StripTagsFilter::class, $filter);
     }
 
     public function testGetFilterForRuleCascade(): void