Small fix for Unittests

Small fix for so unittest don't fail.