Bug fix 0.1