diff --git a/src/Model.js b/src/Model.js
index 9cfb585d..170606d7 100644
--- a/src/Model.js
+++ b/src/Model.js
@@ -382,7 +382,7 @@ export default class Model {
       )
       weightNames = forwardWeightNames.concat(backwardWeightNames)
     } else if (layerClass === 'TimeDistributed') {
-      weightNames = layer.layer.params.map(param => `${layerConfig.name}/${param}`)
+      weightNames = layer.wrappedLayer.params.map(param => `${layerConfig.name}/${param}`)
     } else {
       weightNames = layer.params.map(param => `${layerConfig.name}/${param}`)
     }