Added query to the wrapper