Fix environment tests.

Closes #4.