Revert to os.walk, the tests are failing with the new safe_walk