Fix missing module return