1 files changed, 10 insertions, 1 deletions
diff --git a/www/wiki/extensions/SemanticResultFormats/formats/tree/TreeResultPrinter.php b/www/wiki/extensions/SemanticResultFormats/formats/tree/TreeResultPrinter.php
index d29fed80..a5011d0e 100644
--- a/www/wiki/extensions/SemanticResultFormats/formats/tree/TreeResultPrinter.php
+++ b/www/wiki/extensions/SemanticResultFormats/formats/tree/TreeResultPrinter.php
@@ -137,7 +137,11 @@ class TreeResultPrinter extends ListResultPrinter {
 
 		$this->setQueryResult( null );
 
-		return Html::rawElement( 'div', [ 'class' => 'srf-tree' ], $resultText );
+		if ( trim( $this->params[ 'class' ] ) === '' ) {
+			return $resultText;
+		}
+		
+		return Html::rawElement( 'div', [ 'class' => $this->params[ 'class' ] ], $resultText );
 	}
 
 	/**
@@ -194,6 +198,11 @@ class TreeResultPrinter extends ListResultPrinter {
 			'message' => 'smw-paramdesc-template-arguments',
 		];
 
+		$params['class'] = [
+			'default' => '',
+			'message' => 'srf-paramdesc-class',
+		];
+
 		return $params;
 	}