Oh, I can just use return here