diff --git a/tests/ByteUnitsExtensionIntegrationTest.php b/tests/ByteUnitsExtensionIntegrationTest.php
index ea3fef5..e57e698 100644
--- a/tests/ByteUnitsExtensionIntegrationTest.php
+++ b/tests/ByteUnitsExtensionIntegrationTest.php
@@ -5,9 +5,9 @@
 namespace Marek\Twig\Tests;
 
 use Marek\Twig\ByteUnitsExtension;
-use Twig_Test_IntegrationTestCase;
+use Twig\Test\IntegrationTestCase;
 
-class ByteUnitsExtensionIntegrationTest extends Twig_Test_IntegrationTestCase
+class ByteUnitsExtensionIntegrationTest extends IntegrationTestCase
 {
     protected function getFixturesDir()
     {