quick small fix