Remove dependency on unittest