diff --git a/Controller/CustomFieldController.php b/Controller/CustomFieldController.php
index 8c8e969f9..9eaf354e8 100644
--- a/Controller/CustomFieldController.php
+++ b/Controller/CustomFieldController.php
@@ -7,6 +7,7 @@ use Symfony\Bundle\FrameworkBundle\Controller\Controller;
 use Symfony\Component\Form\Extension\Core\Type\SubmitType;
 
 use Chill\CustomFieldsBundle\Entity\CustomField;
+use Chill\CustomFieldsBundle\Entity\CustomFieldsGroup;
 use Chill\CustomFieldsBundle\Form\CustomFieldType;
 
 
@@ -83,7 +84,7 @@ class CustomFieldController extends Controller
 
         if ($cfGroupId !== null) {
             $cfGroup = $this->getDoctrine()->getManager()
-                  ->getRepository('ChillCustomFieldsBundle:CustomFieldsGroup')
+                  ->getRepository(CustomFieldsGroup::class)
                   ->find($cfGroupId);
             if (!$cfGroup) {
                 throw $this->createNotFoundException('CustomFieldsGroup with id '
@@ -109,7 +110,7 @@ class CustomFieldController extends Controller
     {
         $em = $this->getDoctrine()->getManager();
 
-        $entity = $em->getRepository('ChillCustomFieldsBundle:CustomField')->find($id);
+        $entity = $em->getRepository(CustomField::class)->find($id);
 
         if (!$entity) {
             throw $this->createNotFoundException('Unable to find CustomField entity.');
@@ -127,7 +128,7 @@ class CustomFieldController extends Controller
     {
         $em = $this->getDoctrine()->getManager();
 
-        $entity = $em->getRepository('ChillCustomFieldsBundle:CustomField')->find($id);
+        $entity = $em->getRepository(CustomField::class)->find($id);
 
         if (!$entity) {
             throw $this->createNotFoundException('Unable to find CustomField entity.');