Fixing import error / py3