bugfix add return