Fixed superficial errors in __untie_weights__ and wake_sleep