Update query class
Should not have removed that variable