Small bug fix
Merge pull request #1 from floriniancu/master

Small bug fix