Обновление остатков на складе

Здравствуйте. Работает самописный плагин для обновления остатков. Недавно добавил склад (до этого магазин работал без складов), а остатки по-прежнему обновляются общие. Подскажите, пожалуйста, что поправить в плагине для корректной работы. К сожалению, к автору плагина нет возможности обратиться.

Как я понимаю запись остатков осуществляется с помощью этого:

$model = new shopProductSkusModel();

....

$model->update($sku_id, $count);

Надеюсь на Вашу помощь.