Fixed non-passing test_4.py