Fixed a test as per pmclanahan's comment