diff --git a/tests/utils_diffusion_maps_from_kernel.py b/tests/utils_diffusion_maps_from_kernel.py
index a362a6a8..7505fade 100644
--- a/tests/utils_diffusion_maps_from_kernel.py
+++ b/tests/utils_diffusion_maps_from_kernel.py
@@ -54,7 +54,5 @@ def test_diffusion_maps_eigen(mock_kernel):
 
 def test_determine_multiscale_space(mock_kernel):
     result = diffusion_maps_from_kernel(mock_kernel)
-    determine_multiscale_space(result)
-    assert "EigenValues" in result.keys()
-    assert "EigenValues" in result.keys()
-    assert "T" in result.keys()
+    msresult = determine_multiscale_space(result)
+    assert msresult.shape[0] == result["EigenVectors"].shape[0]