A small fix for the type inference