I wouldn't be surprised to see the remaining A318 follow.
Operating a single-aircraft fleet is a risky operation, and BA's target LCY/JFK market isn't one that will take kindly to last-minute cancellations when the aircraft goes tech with absolutely no prospect of bringing in a substitute.