Quick fix for module error