Add `-p` (`--patch`) and then pick from the diff prompt what parts you want to restore. You can also use this flag with commit.