Minor bug fix  (#306)