test: Fix tests