fixed issue in main test