Fix bug with small-step machine

Small-step machine could not handle environment lookups after evaluating
an expression, since reducing an expression does not return an
environment.