Fixed minor detail that was removed in recent merge