Fix test cases

Update code to use montain/gocheck