add missing module import
Merge pull request #5051 from terminalmage/develop

add missing module import