Small edit to close missing bracket