fixed a refactor whoopsie