diff --git a/Classes/UserFunctions/GridRecords.php b/Classes/UserFunctions/GridRecords.php
index ed97fc4..8679ce2 100644
--- a/Classes/UserFunctions/GridRecords.php
+++ b/Classes/UserFunctions/GridRecords.php
@@ -31,7 +31,13 @@ public function getColPosList(string $content, array $conf, ServerRequestInterfa
     {
         $queryBuilder = GeneralUtility::makeInstance(ConnectionPool::class)->getQueryBuilderForTable('tt_content');
         $currentContentObject = $request->getAttribute('currentContentObject');
-        $pid = $currentContentObject->data['uid'];
+
+        if ($currentContentObject->data['content_from_pid']) {
+            $pid = $currentContentObject->data['content_from_pid'];
+        } else {
+            $pid = $currentContentObject->data['uid'];
+        }
+
         $result = $queryBuilder
             ->select('colPos','pid', 'tx_container_parent')
             ->from('tt_content')