diff --git a/lib/runner.php b/lib/runner.php
index 2d7623d..7ec4581 100644
--- a/lib/runner.php
+++ b/lib/runner.php
@@ -8,6 +8,11 @@
 use phpDocumentor\Reflection\DocBlockFactory;
 use phpDocumentor\Reflection\File\LocalFile;
 use phpDocumentor\Reflection\Php\ProjectFactory;
+use phpDocumentor\Reflection\Php\Class_;
+use phpDocumentor\Reflection\Php\File;
+use phpDocumentor\Reflection\Php\Function_;
+use phpDocumentor\Reflection\Php\Method;
+use phpDocumentor\Reflection\Php\Property;
 
 /**
  * Fixes newline handling in parsed text.
@@ -65,11 +70,9 @@ function get_namespace( $fqsen ) {
 }
 
 /**
- * @param $element
- *
  * @return array
  */
-function export_docblock( $element ) {
+function export_docblock( Property|Method|Hook|File|Function_|Class_ $element ) {
 	$docblock = $element->getDocBlock();
 	if ( ! $docblock ) {
 		return array(