Set-RsRestItemDataSource fails to update linked datasources for Dataset

[jaganmohann246]

Do you want to request a feature or report a bug?
A bug report
What is the current behavior?
If a dataset is configured with a linked datasource, updating the datasource of the DataSet throws an error.
Message is "Error: Cannot validate argument on parameter 'RsItemType'. The argument "DataSet" does not belong to the set "PowerBIReport,Report" specified by the ValidateSet attribute. Supply an argument that is in the set and then try the command again."
If the current behavior is a bug, please provide the steps to reproduce and if possible a minimal demo of the problem.
Need to add 'DataSet' in validateSet in

ReportingServicesTools/ReportingServicesTools/Functions/CatalogItems/Rest/Set-RsRestItemDataSource.ps1

Line 116 in ba9226a

[ValidateSet("PowerBIReport", "Report")]

What is the expected behavior?
Updating ValidateSet in above line will fix this error.
Which versions of Powershell and which OS are affected by this issue? Did this work in previous versions of our scripts?
Not working in v5.1. Not sure of any working versions.