A small update to the previous commit