[Misc] Fix failing test