Adds header file correction to force gravity test main file