Small stupid bug I made during refactoring is fixed