Minor fixes in spatial plotting