Very small fix to @abought changes