Minor bug fix in export_for_test.py