Bug fix v2