fix a nil bug