Fix to work with model node config