FIX: add changes to test_source_estimate