optimized imports
added back an import which was removed but is necessary