Only put query from searchSource into state

When reading the query from a visualizations searchSource, only look for a query on the searchSource itself.