Revert "Fix a bug in sampling from the prior"

This reverts commit 064a3df1c37ade09c6ebc05196f5d61c2fa6d4fe.