FIX: Fix failing test for now