diff --git a/spec/spec-helper.coffee b/spec/spec-helper.coffee
index 799c76e7e..461217c8f 100644
--- a/spec/spec-helper.coffee
+++ b/spec/spec-helper.coffee
@@ -38,7 +38,7 @@ beforeEach ->
   window.project = new Project(specProjectPath)
 
   window.resetTimeouts()
-  atom.packageStates = {}
+  atom.packages.packageStates = {}
   spyOn(atom, 'saveWindowState')
   syntax.clearGrammarOverrides()
   syntax.clearProperties()
diff --git a/src/window.coffee b/src/window.coffee
index 63a21ebe9..323b8bf3e 100644
--- a/src/window.coffee
+++ b/src/window.coffee
@@ -76,7 +76,7 @@ window.unloadEditorWindow = ->
   windowState.set('syntax', syntax.serialize())
   windowState.set('rootView', rootView.serialize())
   atom.deactivatePackages()
-  windowState.set('packageStates', atom.packageStates)
+  windowState.set('packageStates', atom.packages.packageStates)
   atom.saveWindowState()
   rootView.remove()
   project.destroy()