I think in this case you can just return the expression on the indexed subspace i.e. a vector of length 1.