Fix missing module export