Fixing test for 'None'