Small bug fix from previous change
Small bug fix from previous change