Bug fix for dassl.