Fix #352

Fix #352