Fix tiny regression