AWAKE: Fixing a minor problem

Fixing a minor problem