test: work with the current implementation

But is it OK? We should confirm it.