I tried doing that but it worked fine. What version are you trying? Maybe that happens under some special circumstances?
What I did is create a task, assign it to a milestone, go to its edit view, the milestone was already set, saved the task, the milestone was still set. Can you give us more info on how to reproduce this?
Thanks.