Quick fix to #5

This fixes the issue, however later some kind of validator would be probably nicer.