diff --git a/PHPSkills/TrueSkill/FactorGraphTrueSkillCalculator.php b/PHPSkills/TrueSkill/FactorGraphTrueSkillCalculator.php
index 34124a2..98695bd 100644
--- a/PHPSkills/TrueSkill/FactorGraphTrueSkillCalculator.php
+++ b/PHPSkills/TrueSkill/FactorGraphTrueSkillCalculator.php
@@ -52,10 +52,7 @@ class FactorGraphTrueSkillCalculator extends SkillCalculator
         $factorGraph = new TrueSkillFactorGraph($gameInfo, $teams, $teamRanks);
         $factorGraph->buildGraph();
         $factorGraph->runSchedule();
-
-        $test = $factorGraph->getUpdatedRatings();
-
-        // DEBUG: Fix this :)
+        
         $probabilityOfOutcome = $factorGraph->getProbabilityOfRanking();
 
         return $factorGraph->getUpdatedRatings();
diff --git a/PHPSkills/TrueSkill/Layers/IteratedTeamDifferencesInnerLayer.php b/PHPSkills/TrueSkill/Layers/IteratedTeamDifferencesInnerLayer.php
index 39c5b94..eff99ba 100644
--- a/PHPSkills/TrueSkill/Layers/IteratedTeamDifferencesInnerLayer.php
+++ b/PHPSkills/TrueSkill/Layers/IteratedTeamDifferencesInnerLayer.php
@@ -27,6 +27,14 @@ class IteratedTeamDifferencesInnerLayer extends TrueSkillFactorGraphLayer
         $this->_TeamDifferencesComparisonLayer = $teamDifferencesComparisonLayer;
     }
 
+    public function &getLocalFactors()
+    {
+        $localFactors =
+            \array_merge($this->_TeamPerformancesToTeamPerformanceDifferencesLayer->getLocalFactors(),
+                         $this->_TeamDifferencesComparisonLayer->getLocalFactors());
+        return $localFactors;
+    }
+
     public function buildLayer()
     {
         $inputVariablesGroups = &$this->getInputVariablesGroups();