Revert " clean unused import"

This reverts commit dad2ac7f2d2bf1a1b53a3b64f7ac135b4af121fc.
actually clean imports

Signed-off-by: Liew Zhao Wei <zhaoweiliew@gmail.com>