Fix and clean all unit tests