Fix regression that screenshot tests found