Fix bug 2