refactor

Signed-off-by: Akira Hayakawa <ruby.wktk@gmail.com>
fix test

Signed-off-by: Akira Hayakawa <ruby.wktk@gmail.com>