Fix #382

ping @shin- @samalba