Fixed some issues I caused with the tests