Fix #333
Fix #334