Add exposure argument to predict