Keep his leash on. Practice...practice..call him by name to you randomly and praise lavishly with treats or toy..whichever he is most motivated for. Max is very ball motivated, so on off-leash walks he carries a ball and I have one in my pocket. This method has worked very well for us..even called him off a pack of deer.
Your puppy is at "that age!"