Set the correct query param