Fixed some badly written tests