Fix missing module reference