Writing tests for 2.01