Fixed for rails 4 again