Fix tests following source-map update