Fixes in data layer