You are using ``.`` here while for the tests you use full path imports. I have no opinion on this just wanted to point it out :)