Updated test cases to reflect @Param change