diff --git a/src/onehot.jl b/src/onehot.jl
index b1a1a970..33c3921b 100644
--- a/src/onehot.jl
+++ b/src/onehot.jl
@@ -61,5 +61,5 @@ argmax(y::AbstractMatrix, l...) =
 
 # Ambiguity hack
 
-a::TrackedMatrix * b::OneHotVector = TrackedArray(Tracker.Call(*, a, b))
-a::TrackedMatrix * b::OneHotMatrix = TrackedArray(Tracker.Call(*, a, b))
+a::TrackedMatrix * b::OneHotVector = Tracker.track(*, a, b)
+a::TrackedMatrix * b::OneHotMatrix = Tracker.track(*, a, b)