fixed minor feature issue. Fixes #2