Removed an unnecessary import that prevented running on python 2.6