Put the `return` with each of the 2 callbacks instead.