Fixed some testing stuff -ish
Missed a file apparently...
Missed another