Fixes unit tests which wasn't expecting an Error