add missing checks to tests