Updated tests to test the updated database table changes