Fix rake test task