I followed this: https://www.paulligocki.com/installing-calyxos-on-the-google-pixel-6a/#Re-Lock-Boot-Loader

There is still a weird boot screen saying the phone is using a different OS.

Note that locking the boot loader will wipe your data if you already used it.