Minor fix for previous commit

Forgot a file from the last commit