Resolved unused_import warning

Just a small thing.