[BUGFIX] Ensure correct module scope for initialization

Fixes #4