Refactor test case to demonstrate use of the new type param