Start with a combination of the specific movement and general accessories.
Try to fix your problems through volume, intensity, and maybe frequency first.
Be careful of general-specific movements that change the mechanics of the lift significantly, i.e. block pulls. Don’t misdiagnose a technique issue as a strength issue. Your lockout probably isn’t weak; you are just pulling cat-backed and aren’t being honest with yourself about it.