Fix the unittest link error