Uncommenting assertion as it works now