fix bug on previous testing