Small bug fix.