You have it back together and running?
If it's "going away" in the corners, the float may need to be raised a little bit or the oil level safety cutoff is kicking in. does it sound like it's sputtering or just shutting down abruptly? The former would be the fuel level in the bowl, the latter would be the oil safety cutoff.