Add missing 'toolkit' import