Fix #362
Fix #363
Fix #364
Minor fix