and a small fix