small fix to handle put