Slight fix to get rid of "raw types" warnings