Fix python code installation