small import refactoring