https://github.com/probcomp/sppl/blob/efff34fb3d3703247dd7001c36970069c5ac3825/src/spe.py#L53-L54 Also change `memo is False` to `memo in [False, None]`