Fix NoMethodError on NilClass