632ですが、自己解決しました。
関数の戻り値を書けばできました。