Fix test after package path change

Fix failing test after 3795a8