Fix test after package path change

Fix failing test after 3795a80724844d41b30c86833da9f2bc33494d05