Yeah, tests are at 100% again! Corrected import logic for Python3