Forgotten changes to DIA model