Fixed broken tests. Stupidity really.