diff --git a/pdme/model/model.py b/pdme/model/model.py
index 9e062cd..a0ebc1b 100644
--- a/pdme/model/model.py
+++ b/pdme/model/model.py
@@ -89,7 +89,7 @@ class DipoleModel:
 
 				dips.append(tentative_dip)
 			dips_array = pdme.subspace_simulation.sort_array_of_dipoles_by_frequency(dips)
-			tentative_cost = cost_function(numpy.array([dips_array]))
+			tentative_cost = cost_function(numpy.array([dips_array]))[0]
 			if tentative_cost < threshold_cost:
 				chain.append((tentative_cost, dips_array))
 				current = dips_array
diff --git a/pyproject.toml b/pyproject.toml
index 3ea2e36..e30f510 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,7 +29,7 @@ testpaths = ["tests"]
 addopts = "--junitxml pytest.xml --cov pdme --cov-report=xml:coverage.xml --cov-fail-under=50 --cov-report=html"
 junit_family = "xunit1"
 log_format = "%(asctime)s | %(levelname)s | %(pathname)s:%(lineno)d | %(message)s"
-log_level = "DEBUG"
+log_level = "WARNING"
 
 [tool.mypy]
 plugins = "numpy.typing.mypy_plugin"