Added missing files to run tests