add needed import

Signed-off-by: aways <aways@c-rom.org>