lib: Fix tests