Fix: missing a test