Fix #335