Fixed missing import.
Removed redundant code.