Tweak to save B rather than invB