Fix unit test for setQueryTimeout