Fix small error with last commit