Tiny refactor to fix a bug. #99