Update per gabes's feedback