drizzle-orm not inferring type from soft relations

export const PriceTable = pgTable(
    "price",
    {
        id: text("id").primaryKey(),
        productId: text("product_id")
            .notNull()
            .references(() => ProductTable.id),
        active: boolean("active").notNull(),
        currency: text("currency").notNull(),
        unitAmount: integer("unit_amount"),
        createdAt: timestamp("created_at").defaultNow()
    },
    (t) => ({
        productIdIdx: index("idx_price_productId").on(t.productId),
        idIdx: index("idx_price_id").on(t.id)
    })
);

export const priceRelations = relations(PriceTable, ({ one }) => ({
    product: one(ProductTable, {
        fields: [PriceTable.productId],
        references: [ProductTable.id]
    })
}));

export const ProductTable = pgTable("product", {
    id: text("id").primaryKey(),
    active: boolean("active").notNull(),
    name: text("name").notNull(),
    description: text("description"),
    image: text("image"),
    creditAmount: integer("credit_amount").notNull(),
    createdAt: timestamp("created_at").defaultNow(),
    metadata: json("metadata").$type<ProductMetadata>().notNull()
});

export const productRelations = relations(ProductTable, ({ one }) => ({
    price: one(PriceTable)
}));

const price = await ctx.db.query.PriceTable.findFirst({
    where: eq(PriceTable.id, input.priceId),
    with: {
          product: true
        }
});

const price: {
    id: string;
    createdAt: Date | null;
    active: boolean;
    productId: string;
    currency: string;
    unitAmount: number | null;
    product: any;
} | undefined