Update behavior

Update behavior