FIX: Fix test type