Charts as modules - more work

1 files changed, 7 insertions, 4 deletions

diff --git a/app/Module/StatisticsChartModule.php b/app/Module/StatisticsChartModule.php
index d3e9c3c3ca..6e8777b794 100644
--- a/app/Module/StatisticsChartModule.php
+++ b/app/Module/StatisticsChartModule.php
@@ -56,19 +56,22 @@ class StatisticsChartModule extends AbstractModule implements ModuleChartInterfa
 	/**
 	 * Return a menu item for this chart.
 	 *
-	 * @return Menu
+	 * @return Menu|null
 	 */
 	public function getChartMenu(Individual $individual) {
-		global $controller, $WT_TREE;
-		
 		return new Menu(
 			$this->getTitle(), 
-			'statistics.php?ged=' . $WT_TREE->getNameUrl(), 
+			'statistics.php?ged=' . $individual->getTree()->getNameUrl(), 
 			'menu-chart-statistics', 
 			array('rel' => 'nofollow')
 		);
 	}
 	
+	/**
+	 * Return a menu item for this chart - for use in individual boxes.
+	 *
+	 * @return Menu|null
+	 */
 	public function getBoxChartMenu(Individual $individual) {
 		return null;
 	}