Fixed missing module reference