small fix - just error in the unittest