diff --git a/pastasolver/models_handler.py b/pastasolver/models_handler.py
index 6e4daae..d66798a 100644
--- a/pastasolver/models_handler.py
+++ b/pastasolver/models_handler.py
@@ -694,8 +694,7 @@ def get_highest_prob_and_w_id_map(
         max_prob : float = 0.0
         w_id_list : 'list[str]' = []
 
-        for el in current_worlds_dict:
-            w = current_worlds_dict[el]
+        for el, w in current_worlds_dict.items():
             if w.model_query_count > 0 and (w.model_not_query_count == 0 if lower else True):
                 if w.prob == max_prob:
                     max_prob = w.prob
@@ -727,8 +726,7 @@ def get_map_solution(
         else:
             # group by map variables
             map_worlds : 'dict[str,World]' = {}
-            for el in self.worlds_dict:
-                w = self.worlds_dict[el]
+            for el, w in self.worlds_dict.items():
                 if w.model_query_count > 0:
                     # keep both lower and upper
                     sub_w = ModelsHandler.get_sub_world(el, map_id_list)
diff --git a/pastasolver/pasta_solver.py b/pastasolver/pasta_solver.py
index 3e6e1fb..f58d3ee 100644
--- a/pastasolver/pasta_solver.py
+++ b/pastasolver/pasta_solver.py
@@ -596,9 +596,6 @@ def map_inference(self, from_string : str = "") -> 'tuple[float,list[list[str]]]
         self.interface.compute_probabilities()
         max_prob, map_state = self.interface.model_handler.get_map_solution(
             self.parser.map_id_list, self.consider_lower_prob)
-        if self.interface.normalizing_factor >= 1:
-            max_prob = 1
-            # print_warning("No worlds have > 1 answer sets")
 
         if self.normalize_prob and self.interface.normalizing_factor != 0:
             max_prob = max_prob / (1 - self.interface.normalizing_factor)
diff --git a/test/test_approximate_inference.py b/test/test_approximate_inference.py
index c95a737..43a871a 100644
--- a/test/test_approximate_inference.py
+++ b/test/test_approximate_inference.py
@@ -14,6 +14,7 @@ def wrap_test_approximate_inference(self, parameters : utils_for_tests.TestArgum
         args.rejection = parameters.rejection
         args.mh = parameters.mh
         args.gibbs = parameters.gibbs
+        args.approximate_hybrid = False
 
         lp, up = pasta_solver.approximate_solve(args)
 
diff --git a/test/test_map_mpe.py b/test/test_map_mpe.py
index 34f04da..f0d86a2 100644
--- a/test/test_map_mpe.py
+++ b/test/test_map_mpe.py
@@ -1,18 +1,10 @@
 import unittest
-
 import pytest
 
-import importlib.util
-
 import utils_for_tests
 
 from pastasolver.pasta_solver import Pasta
 
-# t_utils = __import__('test_utils')
-
-            
-
-
 class TestClassMapMpe(unittest.TestCase):
 
     def wrap_test_map_mpe(self,
@@ -31,9 +23,8 @@ def wrap_test_map_mpe(self,
             max_p, atoms_list = pasta_solver.upper_mpe_inference()
 
 
-        if max_p is not None and atoms_list is not None:
+        if max_p > 0 and len(atoms_list) > 0:
             self.assertTrue(utils_for_tests.almostEqual(max_p, expected_map_mpe, 5), test_name + ": wrong MAP/MPE")
-    
             self.assertTrue(utils_for_tests.check_if_lists_equal(atoms_list, expected_atoms_list), test_name + ": wrong atoms list")