diff --git a/Classes/Hooks/Tca/AddFilesToSelector.php b/Classes/Hooks/Tca/AddFilesToSelector.php
index 8c6515b..3360de9 100644
--- a/Classes/Hooks/Tca/AddFilesToSelector.php
+++ b/Classes/Hooks/Tca/AddFilesToSelector.php
@@ -33,7 +33,7 @@ class AddFilesToSelector {
 	function main(&$params,&$pObj)	{
 
 		// get the current page ID
-		$thePageId = $params['row']['pid'];
+		$thePageId = $params['flexParentDatabaseRow']['pid'];
 
 		/** @var TemplateService $template */
 		$template = GeneralUtility::makeInstance('TYPO3\\CMS\\Core\\TypoScript\\TemplateService');
@@ -89,4 +89,4 @@ function main(&$params,&$pObj)	{
 			}
 		}
 	}
-}
\ No newline at end of file
+}