fix unit tests while i'm here