FIX: Fix test running