diff --git a/src/TracedRange.jl b/src/TracedRange.jl
index 6da8ccf812..657fb4f711 100644
--- a/src/TracedRange.jl
+++ b/src/TracedRange.jl
@@ -177,13 +177,13 @@ function Base._reshape(parent::TracedUnitRange, dims::Dims)
     return Base.__reshape((parent, IndexStyle(parent)), dims)
 end
 
-function (C::Base.Colon)(start::TracedRNumber{T},stop::TracedRNumber{T}) where T
-    TracedUnitRange(start,stop)
+function (C::Base.Colon)(start::TracedRNumber{T}, stop::TracedRNumber{T}) where {T}
+    return TracedUnitRange(start, stop)
 end
-function (C::Base.Colon)(start::TracedRNumber{T},stop::T) where T
-    C(start,TracedRNumber{T}(stop))
+function (C::Base.Colon)(start::TracedRNumber{T}, stop::T) where {T}
+    return C(start, TracedRNumber{T}(stop))
 end
-function (C::Base.Colon)(start::T,stop::TracedRNumber{T}) where T
-    C(TracedRNumber{T}(start),stop)
+function (C::Base.Colon)(start::T, stop::TracedRNumber{T}) where {T}
+    return C(TracedRNumber{T}(start), stop)
 end
 end
diff --git a/test/ranges.jl b/test/ranges.jl
index 949bb00627..93ab86ed1c 100644
--- a/test/ranges.jl
+++ b/test/ranges.jl
@@ -1,9 +1,9 @@
-using Reactant,Test
+using Reactant, Test
 
 @testset "ranges" begin
-    i = Reactant.to_rarray(5,track_numbers=true)
+    i = Reactant.to_rarray(5; track_numbers=true)
     @test Array{Int64}(@jit(1:i)) == collect(1:5)
     @test Array{Int64}(@jit(i:10)) == collect(5:10)
-    j = Reactant.to_rarray(10,track_numbers=true)
-    @test Array{Int64}(@jit(i:j)) == collect(5:10) 
+    j = Reactant.to_rarray(10; track_numbers=true)
+    @test Array{Int64}(@jit(i:j)) == collect(5:10)
 end