Fixed  return bug.