diff --git a/diefpy/dief.py b/diefpy/dief.py
index 100a84e..ffa6830 100644
--- a/diefpy/dief.py
+++ b/diefpy/dief.py
@@ -296,8 +296,8 @@ def performance_of_approaches_with_dieft(traces: np.ndarray, metrics: np.ndarray
             if a not in np.unique(dieft_res['approach']):
                 continue
 
-            dieft_ = dieft_res[dieft_res['approach'] == a]['dieft'][0]
-            submetric = metrics[metrics['approach'] == a]
+            dieft_ = dieft_res[(dieft_res['approach'] == a) & (dieft_res['test'] == t)]['dieft'][0]
+            submetric = metrics[(metrics['approach'] == a) & (metrics['test'] == t)]
 
             throughput = submetric['comp'][0] / submetric['totaltime'][0]
             invtfft = 1 / submetric['tfft'][0]
diff --git a/setup.py b/setup.py
index ec226ed..cff1d34 100644
--- a/setup.py
+++ b/setup.py
@@ -11,13 +11,13 @@ def readme():
 
 
 setup(name='diefpy',
-      version='1.0.3',
+      version='1.0.4',
       packages=['diefpy'],
       license='MIT',
       author='Philipp D. Rohde, Nikoleta Themeliotou',
       author_email='philipp.rohde@tib.eu',
       url='https://github.com/SDM-TIB/diefpy',
-      download_url='https://github.com/SDM-TIB/diefpy/archive/refs/tags/v1.0.3.tar.gz',
+      download_url='https://github.com/SDM-TIB/diefpy/archive/refs/tags/v1.0.4.tar.gz',
       description='Python package for computing diefficiency metrics dief@t and dief@k.',
       long_description=long_description,
       long_description_content_type="text/markdown",