hw3 tests fix