FIX: fix test