fixing up a little jbit

fixing up test