我写的代码应该给消息的作者(我是蓝色的)一个角色，它不工作，请重写我的代码

// 重写前的代码
class Message {
  constructor(author) {
    this.author = author;
    this.role = null;
  }

  getAuthorRole() {
    return this.role;
  }

  setAuthorRole(role) {
    this.role = role;
  }
}

// 重写后的代码
class Message {
  constructor(author) {
    this.author = author;
    this.role = "工作角色"; // 修改默认角色为工作角色，确保作者角色工作正常
  }

  getAuthorRole() {
    return this.role;
  }

  setAuthorRole(role) {
    this.role = role;
  }
}

// 使用示例
const message = new Message("蓝色的作者");
console.log(message.getAuthorRole()); // 输出："工作角色"
message.setAuthorRole("其他角色");
console.log(message.getAuthorRole()); // 输出："其他角色"