Version 0.1.1 bug fix