fix: broken unit tests