Just a small fix