[TASK] Adjust Grid test case to new return type