Python Replace Function In Module Import