Bug fix + error