As long as you're doing something productive, I think that's all that really matters. Personally, I would go to other institutions just so you get a sense of the world and not just your own backyard. If working with your PI long term leads to a publication, it is probably better to stay though.