If they can't find better way to show progress and improvement from Suraj's side, then OK bring back his memory. But please makers make it quick. Before some disaster idea is born in your smart heads.
Since its childhood memory I sincerely hope it will be just to give us and Chakor hope. To make Suraj to change his way of looking at some things. To end this trip (maybe) on light note. Something that will be cause of slight doubt in Suraj. Doubt which will grow everytime he looks at Chakor, women who loves him, but his Malkin doesn't want to give him right explanation. I would love to see Suraj after he realises that he knows Chakor from childhood to wonder why his Malkin lied to him that the photo Chakor give him is fake one. But I know that makers already forgot that. I would love to see Suraj asking Imli does he know Chakor from childhood and she obviously lying to him after he remembers something. I don't want him to go against Hyar just like that. I want him to doubt, to suspect and then in right moment to go against them. But even if they right now show him going against Hyar it won't be bad. Just that I would prefer first option. And it gives them opportunity to dragg a little more.
I'm still and always will be that opinion that he should get his memory back after he changes to completly good person (he is good even now, just he doesnt know how and why to show that because only thing he is consious is folloving orders) and ask forgiveness from Chakor. But some bits of memory, flashes or childhood memory's can be brought in story, I'm OK with that. Last time he changed with his childhood memories. It can be now too. What he should not remember right now is period after Sukor fell in love, more then that he should not remember anything after 10 years leap as of yet.