Revert code = nil fix