fix return keyword bug

tiny change