Fix unittest import

Miau