Fix issue where weights were not correctly added to ds

Fix issue where weights were not correctly added to ds #154