bug fix and add simple test