added a fix for pit testing