need to break the loop later when action is None (#155)

need to break the loop later when action is None (#155) #348