Bug fix v0.1