fixed issue in test files