Issue: #5

  * Minor bug fix