Remove extra import

Signed-off-by: Sol Jerome <sol.jerome@gmail.com>