fixed more test error