Fix python module references