fix(tests): fix tests due to change in query fields methods behavior