Fix #308
Fix #320