Fix nil reference