Bug fix for previous commit

Signed-off-by: Scott Wilson <i@scottwilson.ca>