Fix build do to missing module import.