diff --git a/ModelDescriber/JMSModelDescriber.php b/ModelDescriber/JMSModelDescriber.php
index ce099a6..42d28a7 100644
--- a/ModelDescriber/JMSModelDescriber.php
+++ b/ModelDescriber/JMSModelDescriber.php
@@ -100,8 +100,15 @@ class JMSModelDescriber implements ModelDescriberInterface, ModelRegistryAwareIn
                     continue;
                 }
 
+                if(!array_key_exists($name, $model->getGroups()) || !is_array($model->getGroups()[$name])) {
+                    $groups = $model->getGroups();
+                } else {
+                    $groups = $model->getGroups()[$name];
+                }
+
+
                 $property->setRef(
-                    $this->modelRegistry->register(new Model(new Type(Type::BUILTIN_TYPE_OBJECT, false, $type), $model->getGroups()))
+                    $this->modelRegistry->register(new Model(new Type(Type::BUILTIN_TYPE_OBJECT, false, $type), $groups))
                 );
             }
 
diff --git a/Tests/Functional/Controller/JMSController.php b/Tests/Functional/Controller/JMSController.php
index 3aa6dd6..94bf5d8 100644
--- a/Tests/Functional/Controller/JMSController.php
+++ b/Tests/Functional/Controller/JMSController.php
@@ -13,6 +13,7 @@ namespace Nelmio\ApiDocBundle\Tests\Functional\Controller;
 
 use Nelmio\ApiDocBundle\Annotation\Model;
 use Nelmio\ApiDocBundle\Tests\Functional\Entity\JMSUser;
+use Nelmio\ApiDocBundle\Tests\Functional\Entity\JMSComplex;
 use Nelmio\ApiDocBundle\Tests\Functional\Entity\VirtualProperty;
 use Sensio\Bundle\FrameworkExtraBundle\Configuration\Route;
 use Swagger\Annotations as SWG;
@@ -42,4 +43,16 @@ class JMSController
     public function yamlAction()
     {
     }
+
+    /**
+     * @Route("/api/jms_complex", methods={"GET"})
+     * @SWG\Response(
+     *     response=200,
+     *     description="Success",
+     *     @Model(type=JMSComplex::class, groups={"list", "details", "user" : {"list"}})
+     * )
+     */
+    public function complexAction()
+    {
+    }
 }
diff --git a/Tests/Functional/Entity/JMSComplex.php b/Tests/Functional/Entity/JMSComplex.php
new file mode 100644
index 0000000..3cb6006
--- /dev/null
+++ b/Tests/Functional/Entity/JMSComplex.php
@@ -0,0 +1,43 @@
+<?php
+/**
+ * Created by PhpStorm.
+ * User: Frisks
+ * Date: 2/12/2017
+ * Time: 1:18 PM
+ */
+
+namespace Nelmio\ApiDocBundle\Tests\Functional\Entity;
+
+use JMS\Serializer\Annotation as Serializer;
+
+/**
+ * Class JMSComplex
+ * @package Nelmio\ApiDocBundle\Tests\Functional\Entity
+ * @Serializer\ExclusionPolicy("all")
+ */
+class JMSComplex
+{
+
+    /**
+     * @Serializer\Type("integer")
+     * @Serializer\Expose
+     * @Serializer\Groups({"list"})
+     */
+    private $id;
+
+
+    /**
+     * @Serializer\Type("Nelmio\ApiDocBundle\Tests\Functional\Entity\JMSUser")
+     * @Serializer\Expose
+     * @Serializer\Groups({"details"})
+     */
+    private $user;
+
+    /**
+     * @Serializer\Type("string")
+     * @Serializer\Expose
+     * @Serializer\Groups({"list"})
+     */
+    private $name;
+
+}
diff --git a/Tests/Functional/Entity/JMSUser.php b/Tests/Functional/Entity/JMSUser.php
index ba5c77d..db0e760 100644
--- a/Tests/Functional/Entity/JMSUser.php
+++ b/Tests/Functional/Entity/JMSUser.php
@@ -24,6 +24,7 @@ class JMSUser
     /**
      * @Serializer\Type("integer")
      * @Serializer\Expose
+     * @Serializer\Groups({"list"})
      *
      * @SWG\Property(description = "User id", required = true, readOnly = true, title = "userid", example=1, default = null)
      */
@@ -41,8 +42,8 @@ class JMSUser
     /**
      * @Serializer\Type("string")
      * @Serializer\Expose
-     *
      * @SWG\Property(readOnly = false)
+     * @Serializer\Groups({"details"})
      */
     private $email;
 
diff --git a/Tests/Functional/JMSFunctionalTest.php b/Tests/Functional/JMSFunctionalTest.php
index c73c1ad..20e0bcf 100644
--- a/Tests/Functional/JMSFunctionalTest.php
+++ b/Tests/Functional/JMSFunctionalTest.php
@@ -63,6 +63,31 @@ class JMSFunctionalTest extends WebTestCase
         ], $this->getModel('JMSUser')->toArray());
     }
 
+    public function testModelComplexDocumentation()
+    {
+        $this->assertEquals([
+            'type' => 'object',
+            'properties' => [
+                              'id' => ['type' => 'integer'],
+                'user' => ['$ref' => '#/definitions/JMSUser2'],
+                'name' => ['type' => 'string']
+            ],
+        ], $this->getModel('JMSComplex')->toArray());
+
+        $this->assertEquals([
+            'type' => 'object',
+            'properties' => [
+                'id' => [
+                    'type' => 'integer',
+                    'title' => 'userid',
+                    'description' => 'User id',
+                    'readOnly' => true,
+                    'example' => '1',
+                ],
+            ],
+        ], $this->getModel('JMSUser2')->toArray());
+    }
+
     public function testYamlConfig()
     {
         $this->assertEquals([
diff --git a/composer.json b/composer.json
index a3466bf..de9cc69 100644
--- a/composer.json
+++ b/composer.json
@@ -34,6 +34,7 @@
         "symfony/browser-kit": "^2.8|^3.0|^4.0",
         "symfony/cache": "^3.1|^4.0",
         "symfony/phpunit-bridge": "^3.3",
+        "symfony/stopwatch": "^2.8|^3.0|^4.0",
         "sensio/framework-extra-bundle": "^3.0",
         "symfony/stopwatch": "^2.8|^3.0|^4.0",
         "doctrine/annotations": "^1.2",