:green_heart: Fix tests