Fix module import to enable new scripts work for both python 2.7 and 3.3