FIX: fix broken test