Fixed a bug in the bugfix -.-