bug fix
bug fix to make it work