Fix #376 #377

cc @drj11