Minor bug fix for previous commit