Olympe script stops working after (10?) seconds

I have a script that uses the followme in Olympe. It works perfectly fine, but the drone just stops moving after 5 or so seconds. It does this reguardless of timeout settings or if I remove the .disconnect. Here is my code - any thoughts?

# -*- coding: UTF-8 -*-

import olympe
from olympe.messages.ardrone3.Piloting import TakeOff, moveBy, Landing
from olympe.messages.ardrone3.PilotingState import FlyingStateChanged
from olympe.messages import follow_me
from olympe.messages.follow_me import start,stop,state,set_target_is_controller,target_is_controller
drone = olympe.Drone(“10.202.0.1”)
drone.connect()
drone(
TakeOff()
>> FlyingStateChanged(state=“hovering”, _timeout=60)
).wait()
drone(
follow_me.set_target_is_controller(1, _timeout=60)
).wait()
drone(
follow_me.start(mode=“leash”, _timeout=60)
#>> state(mode=“leash”, behaviour=“follow”)
).wait()
drone(Landing()).wait()
#drone.disconnect()

I think you still need to send PING/PONG in order for the connection to remain open.
Otherwise it will get closed after 5 seconds