Fixed Agent stopping on completion