Generator fix for rails 4