I think it could definitely also be "Die Tochter hatte einfach den Partner behalten." There's no reason einfach has to be after Partner.
You're right.