Bugfixes! \o/