Fix module

This fixes #2