Fixes the Bug

Fixes #2