Fix API signature