Forgot to fix it in test also