Nevermind, missed the `return` above.