fixing the build: fixed test according recent change