Minor bug correction v1.0.1.1