- Fixed nil reference error