fix the nil bug