Fix code typo (missing brackets)