Fix assertion method call

Follow up to #50382.

`assert_queries` was renamed to `assert_queries_count` in #50373.