Restore the if else in the puller