bugfix and add test module

bugfix and add test module