Removed debug code from unit test