diff --git a/seminar/models/tvorba.py b/seminar/models/tvorba.py
index 00d1adfd..5d35c2ae 100644
--- a/seminar/models/tvorba.py
+++ b/seminar/models/tvorba.py
@@ -480,6 +480,7 @@ class Problem(SeminarModelBase,PolymorphicModel):
 	def admin_url(self):
 			return reverse('admin:seminar_problem_change', args=(self.id, ))
 
+	@cached_property
 	def hlavni_problem(self):
 		""" Pro daný problém vrátí jeho nejvyšší nadproblém."""
 		return hlavni_problem(self)
diff --git a/vysledkovky/utils.py b/vysledkovky/utils.py
index 5a22a447..ce988c50 100644
--- a/vysledkovky/utils.py
+++ b/vysledkovky/utils.py
@@ -286,19 +286,13 @@ class VysledkovkaCisla(Vysledkovka):
 				for problem in self.podproblemy[-1]
 		}
 
-		# Mapování podproblémů na problémy
-		problem2nadproblem = dict()
-		for tema in self.temata_a_spol:
-			for problem in self.podproblemy[tema.id]:
-				problem2nadproblem[problem.id] = tema.id
-
 		# Získáme query všech sčítaných hodnocení
 		hodnoceni_do_cisla = self.hodnoceni_do_cisla
 
 		# Sečteme hodnocení
 		for hodnoceni in hodnoceni_do_cisla:
 			prob = hodnoceni.problem
-			nadproblem = problem2nadproblem[prob.id]
+			nadproblem = prob.hlavni_problem.id
 
 			problem_slovnik = body_za_problemy[nadproblem][prob.id]
 			nadproblem_slovnik = body_za_temata[nadproblem]