Removed unused import that PyCharm added