Add tests back to Rakefile