Remove now superfluous argument from predict_b_extend. Change-Id: I7a76756842af9ce806c6e0e1f98f294af748e8bd