Fix rake test