Fixed a bug!

Fixes #14