Fix small oversight