A small fix