Remove unittest instead of sure