Fix issue #2 (bug)