Fix code to match tests
Fix code to match tests

aequitas/isort,tomviner/isort,leepa/isort,jamesbroadhead/isort,adamchainz/isort,aequitas/isort,jamesbroadhead/isort,adamchainz/isort,PyCQA/isort,levic/isort,gforcada/isort,jmullan/isort,pahko/isort,graingert/isort,AbsoluteMSTR/isort,tomviner/isort,pahko/isort,gforcada/isort,AbsoluteMSTR/isort,jmullan/isort,wimglenn/isort,levic/isort,PyCQA/isort,graingert/isort,leepa/isort,wimglenn/isort