Fixed errors with rake test