Add suggested fix for simple case