Forget: fixed minor bug

fixed minor bug