FIX: Fix test