fix a bug

change return value