Fix testing (for real this time i promise)