fixed minor feature issue. Fixes #2

fixed minor feature issue