Added code to avoid errors in test code