[BUGFIX] Fix Unit Tests

Releases: 1.0.0
Fixes: #50731