Controlling lock up in a 904 lock up

Different spring for the lockup valve would do that.
I seem to remember a kit, but you would have to do some research on it.
Does your VB look like this then?

The circled section is the lockup valve assy.