Fixed a stupid 'refactoring'