My friend came into work to see me and we chatted for 30 mins, now Im worried the bosses will sack me?

Twice in the last fortnight my friend has came down late to the store that I work to chat to me. The first time one of the managers saw us and told me to get some work done, I told her that I was taking the time I talked instead of going for my break. My friend came down to see me yesterday and I lost track of time and noticed when she left that we had been chatting for 30 mins! Now Im worried the bosses will sack me. What can I do? I cant exactly tell my friend to stop coming to see me incase she takes offense :(