Updating tests to include case