Slight changes to my tests