small change

added import for py2exe to work properly