ちょっと何いってるかよくわからないのだが、returnつけておけばそのコードでいいのでは。