bug fix
bug fix & test