Fix one more bug