Fixed missing files in last commit