Tweak test.jl file