fixed issue in test file