Fixed minor regression