Small bug correction.