Fix #207 & #209