Change test in Rakefile