FIX: fixed bug due to parallelism in softmax action selection