V 0.5.1 bug fix