Added missing check in test.