Quick fix of broken function
Merge pull request #119 from dyslaw-pesail/master

Quick fix of broken function