FIX: Fix type