fix problem with return value