Fix #334