Fix error for rake test