diff --git a/src/Payum/YiiExtension/PayumComponent.php b/src/Payum/YiiExtension/PayumComponent.php
index <HASH>..<HASH> 100644
--- a/src/Payum/YiiExtension/PayumComponent.php
+++ b/src/Payum/YiiExtension/PayumComponent.php
@@ -1,6 +1,8 @@
 <?php
 namespace Payum\YiiExtension;
 
+\Yii::import('Payum\YiiExtension\TokenFactory', true);
+
 use Payum\Core\PaymentInterface;
 use Payum\Core\Registry\RegistryInterface;
 use Payum\Core\Registry\SimpleRegistry;
Add required code to include TokenFactory in Yii