diff --git a/test/mass_matrix.jl b/test/mass_matrix.jl
index cead2e8e6..862e3ec81 100644
--- a/test/mass_matrix.jl
+++ b/test/mass_matrix.jl
@@ -12,8 +12,8 @@ using Test
         return nothing
     end
     M = [
-        1.0 0 0
-        0 1.0 0
+        1 0 0
+        0 1 0
         0 0 0
     ]
     f = ODEFunction(rober, mass_matrix=M)