bug fix
fix test