you should also remove the pytest import.