Actually fix #186. This is what I get for writing code while sleep deprived.