Fix internal tests