Cleaned testing code and improved assertion of wrapped element