Fix query bugs