Fixed wrong import of unittest module