Fix/#14

fix the failed build issue and update the test