Fix to a small fix