Fixed that bug

Closes #1