Made the example return the right queriable type