changes made to fix tests