Little fix to APi change