Fix tests for Rails 3.0