Fix for swift unit test