Fixed a little refactoring issue

Oops I did it again
https://www.youtube.com/watch?v=CduA0TULnow&spfreload=10