fixed some small problems for main_test.py