Update the 'test' to check for null