Minor bug in last commit