If it's mildly enlarged it absolutely can go back to normal if the cause is successfully treated. The heart is a muscle and as you likely know, muscles enlarge or atrophy based upon how hard it's used. I've gotten the impression that you're quite young so I think that bodes well for a full recovery.