Fixed unit tests for WPF version.