Fixed tests for rails 4.1