refactor: directly expose KeyModule