Corrected the import command in run_cosmicx_test.py