Added tests for source recovery