A small fix 2