diff --git a/test/functional/timelog_custom_fields_visibility_test.rb b/test/functional/timelog_custom_fields_visibility_test.rb
index 5361ee03e..f7633fe3a 100644
--- a/test/functional/timelog_custom_fields_visibility_test.rb
+++ b/test/functional/timelog_custom_fields_visibility_test.rb
@@ -95,8 +95,8 @@ class TimelogCustomFieldsVisibilityTest < ActionController::TestCase
     p1 = Project.generate!
     p2 = Project.generate!
     user = User.generate!
-    User.add_to_project(user, p1, Role.find_all_by_id([1, 3]))
-    User.add_to_project(user, p2, Role.find_all_by_id(3))
+    User.add_to_project(user, p1, Role.where(:id => [1, 3]).all)
+    User.add_to_project(user, p2, Role.where(:id => 3).all)
     TimeEntry.generate!(
       :issue => Issue.generate!(:project => p1, :tracker_id => 1,
                                 :custom_field_values => {@field2.id => 'ValueA'}))