Fixing test case
fixing up the lib