Fixed error with passing nil to unit.find