Fix test paths
PEP8 fixes