Add missing helper module imports