Bug Fix

Fixes #97, probably not the best way to handle it