Pass only the first element so the dimension of the prior is correct after returning.
Fixes #80
Passes check and tests