Fixed slight refactoring bug