Fix test to reflect change by @Psycojoker