You put 6bb in preflop with AK and lost 40bb of your stack once hitting top pair. There's absolutely nothing wrong with that play at all.

If he had 100bb stack and pushes all in to pot size bet on turn then you should perhaps get away but not here.