Remove with from unittest