Small mapping fix