Fix nil bug