Add tests to check for db column