diff --git a/test/functional/custom_fields_controller_test.rb b/test/functional/custom_fields_controller_test.rb
index ab051a6..0ef0acd 100644
--- a/test/functional/custom_fields_controller_test.rb
+++ b/test/functional/custom_fields_controller_test.rb
@@ -45,7 +45,7 @@ class CustomFieldsControllerTest < ActionController::TestCase
     get :new
     assert_response :success
     assert_template 'select_type'
-    assert_select 'input[name=type]', CustomField.subclasses.size
+    assert_select 'input[name=type]', CustomFieldsHelper::CUSTOM_FIELDS_TABS.size
     assert_select 'input[name=type][checked=checked]', 1
   end