diff --git a/test/basic.jl b/test/basic.jl
index 37176b89..93d77275 100644
--- a/test/basic.jl
+++ b/test/basic.jl
@@ -10,7 +10,7 @@ d = Affine(10, 20)
 
 d1 = @net x -> x * d.W + d.b
 
-Flux.infer(d, (1, 10))
+@test Flux.infer(d, (1, 10)) == (1,20)
 
 # Skip this before new DataFlow is released.
 # let