fix unit t tests