Fix Lib Bug

Fix Lib bug