fix: only one, direct return (#3317)
fix: only one, direct return (#3317)

Co-authored-by: hhstuhacker <e6948b00423af86f5083a430a0a3945bd05f414f@gmail.com>
tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs,tensorflow/tfjs