Added in a `return` call. Was adding this from the old code and forgot to do flow control on it...