Update tests for new column