Resolved unused_import warning