Ensure class name is set - for rails 4