Fix regression nickm pointed out