MOD: fix bug