Fixed unit test which used assertIsNotNone