keeping my mind focused on studies while having a boyfriend

so recently ive been together with this boy, and while we’re all good and everything, ive been noticing that my mind keeps drifting off towards him even when he isnt around and this has caused me to be less focused when im studying.

ive also been checking my phone and thinking more abt when he’s going to reply, and even if i put my phone away i still keep thinking about it

this year is my A levels year and i seriously need help to be more focused. do you all have any tips on how to remain more focused?