fixed a missing return at the end of test_4 file