Bug fix 2