Quick bug fix related to #30.