Fix a couple of assignment 3 tests