fix bug on test