fixed small thing

need to return the query