Fixed issue with missing files