رفع مشکل #18

Fixes #18