little fix was erroring out in test mode