Fix for rails 4?