Fix minor bug in setup  Fixes Issue #77