Fix nil logic