diff --git a/test/unit/helpers/custom_fields_helper_test.rb b/test/unit/helpers/custom_fields_helper_test.rb
index f353c2aa9..1965874b8 100644
--- a/test/unit/helpers/custom_fields_helper_test.rb
+++ b/test/unit/helpers/custom_fields_helper_test.rb
@@ -33,7 +33,7 @@ class CustomFieldsHelperTest < ActionView::TestCase
     value = CustomValue.new(:value => 'bar', :custom_field => field)
     field.id = 52
 
-    assert_equal '<input id="object_custom_field_values_52" name="object[custom_field_values][52]" type="text" value="bar" />',
+    assert_equal '<input class="foo_cf" id="object_custom_field_values_52" name="object[custom_field_values][52]" type="text" value="bar" />',
       custom_field_tag('object', value)
   end
 
@@ -41,7 +41,7 @@ class CustomFieldsHelperTest < ActionView::TestCase
     field = CustomField.new(:field_format => 'foo')
     field.id = 52
 
-    assert_equal '<input id="object_custom_field_values_52" name="object[custom_field_values][52]" type="text" value="" />',
+    assert_equal '<input class="foo_cf" id="object_custom_field_values_52" name="object[custom_field_values][52]" type="text" value="" />',
       custom_field_tag_for_bulk_edit('object', field)
   end
 end