lib fix + unittest