Fluent API doesn't work as expected with client extensions

client.$extends({
  query: {
    $allModels: {
      async $allOperations({ args, query }) {
        const user = store.get('user');
        const userType = store.get('userType');
        
        let configSql: Sql;
        
        switch (userType) {
          case UserTypeEnum.ADMIN:
            configSql = Prisma.sql`SELECT set_config('app.bypass_rls', 'on', TRUE)`;
            break;
          case UserTypeEnum.CUSTOMER:
            configSql = Prisma.sql`SELECT set_config('app.current_customer_id', ${user.id}, TRUE)`;
            break;
        }
        
        const [, result] = await client.$transaction([
          client.$executeRaw(configSql),
          query(args),
        ]);
        return result;
      },
    },
  },
});

@ResolveField(() => User)
  private async author(@Parent() { id }: BlogPost): Promise<User> {
    return this._prisma.blogPost.findUnique({ where: { id }}).author();
  }