Fixed issue #39

Păi modifică și aici, te rog (nu știu cum renunți la pull requestul ăsta ca să faci altul, sper că te prinzi tu).