TODO: convert test to TOX