This is the perfect ending to that ep. You keep the important character growth and actualization there for Martha when she leaves, but you have the Doctor coming back for her so that they can continue their awesome space adventures. The best of both worlds.