- small fix to unit tests