FIX: unit test was using old API