Fixing files so rake test passes