small fix for test error.