Fix the wrap string + add tests

Change-Id: I93b9c43d2346033e2c7dbd79cff90b7809ead337