Small bug fix.

Update