Fixed source_test.py