First you’re aware of the thinking; then you have some realization thoughts and continue to think because you are happy that you have some realization. Then awareness is gone.
So, try to be aware of whatever is happening – wholesome thought or unwholesome thought arising, you need to continue to recognize it.
Continue to be aware that you’re thinking. That’s it.