Fix source-map issue