Added missing return statement
Merge pull request #36 from diskokeks/0.5.1

Add missing return statement