small fix v2