Not a native speaker here but only the latter sounds ok to me. I'd probably still use "when" instead of "after" there (not that there's anything wrong with "after") I also think that the first sentence would be fine if you just replaced "after" with "when".
New words, one handy idiom, and a 2-minute quiz — delivered to your inbox to keep your streak alive.