Forgot to add something to the lib, fixed