diff --git a/Project.toml b/Project.toml
index 30a0eff2d..9296ef23a 100644
--- a/Project.toml
+++ b/Project.toml
@@ -2,7 +2,7 @@ name = "OpticSim"
 uuid = "24114763-4efb-45e7-af0e-cde916beb153"
 authors = ["Brian Guenter", "Charlie Hewitt", "Ran Gal", "Alfred Wong"]
 repository = "https://github.com/microsoft/OpticSim.jl"
-version = "0.5.0-DEV"
+version = "0.5.0"
 
 [deps]
 CSV = "336ed68f-0bac-5ca0-87d4-7b16caf5d00b"
diff --git a/src/Optical/OpticalSystem.jl b/src/Optical/OpticalSystem.jl
index 78e4dc18c..33853fed5 100644
--- a/src/Optical/OpticalSystem.jl
+++ b/src/Optical/OpticalSystem.jl
@@ -597,14 +597,14 @@ function traceMT(
                         print("\rTraced: ~ $t / $(length(sources))        Elapsed: $(dif)s        Left: $(left)s           ")
                     end
                 end
-                trace(system, generateray(sources, k), test = test)
+                trace(system, sources[k]; test)
             end
         else
             for k in f:l
                 if k % update_timesteps == 0
                     Threads.atomic_add!(total_traced, update_timesteps)
                 end
-                trace(system, generateray(sources, k), test = test)
+                trace(system, sources[k]; test)
             end
         end
     end
@@ -713,7 +713,7 @@ function tracehitsMT(
                         print("\rTraced: ~ $t / $(length(sources))        Elapsed: $(dif)s        Left: $(left)s           ")
                     end
                 end
-                lt = trace(system, generateray(sources, k), test = test)
+                lt = trace(system, sources[k]; test)
                 if lt !== nothing
                     push!(results[i], lt)
                 end
@@ -723,7 +723,7 @@ function tracehitsMT(
                 if k % update_timesteps == 0
                     Threads.atomic_add!(total_traced, update_timesteps)
                 end
-                lt = trace(system, generateray(sources, k), test = test)
+                lt = trace(system, sources[k]; test)
                 if lt !== nothing
                     push!(results[i], lt)
                 end