make it possible to change query class