adding a small fix