Fix: Fix test