Fixed issue with not including files in unit test file.