diff --git a/test/rulesets/Base/arraymath.jl b/test/rulesets/Base/arraymath.jl
index 7c76aa17a..0a1416444 100644
--- a/test/rulesets/Base/arraymath.jl
+++ b/test/rulesets/Base/arraymath.jl
@@ -215,8 +215,8 @@
         @gpu test_frule(+, randn(2), randn(2), randn(2))
         # rev
         @gpu test_rrule(+, randn(4, 4), randn(4, 4), randn(4, 4))
-        @gpu test_rrule(+, randn(3), randn(3,1), randn(3,1,1))
-        test_rrule(+, randn(3,3), Diagonal(randn(3)), randn(3,3,1))
-        test_rrule(+, randn(3,3), Diagonal(randn(3)), Symmetric(randn(3,3)))
+        @gpu test_rrule(+, randn(3), randn(3, 1), randn(3, 1, 1))
+        test_rrule(+, randn(3, 3), Diagonal(randn(3)), randn(3, 3, 1))
+        test_rrule(+, randn(3, 3), Diagonal(randn(3)), Symmetric(randn(3, 3)))
     end
 end