1 files changed, 4 insertions, 1 deletions

diff --git a/drivers/video/ps3fb.c b/drivers/video/ps3fb.c
index 13c876c6e32c..1baa1c93a224 100644
--- a/drivers/video/ps3fb.c
+++ b/drivers/video/ps3fb.c
@@ -1123,7 +1123,7 @@ static int __devinit ps3fb_probe(struct ps3_system_bus_device *dev)
 
         info = framebuffer_alloc(sizeof(struct ps3fb_par), &dev->core);
         if (!info)
-                goto err_context_unmap;
+                goto err_context_fb_close;
 
         par = info->par;
         par->mode_id = ~ps3fb_mode;     /* != ps3fb_mode, to trigger change */
@@ -1188,6 +1188,8 @@ err_fb_dealloc:
         fb_dealloc_cmap(&info->cmap);
 err_framebuffer_release:
         framebuffer_release(info);
+err_context_fb_close:
+        lv1_gpu_fb_close(ps3fb.context_handle);
 err_context_unmap:
         lv1_gpu_context_iomap(ps3fb.context_handle, GPU_IOIF, xdr_lpar,
                               ps3fb_videomemory.size, CBE_IOPTE_M);
@@ -1233,6 +1235,7 @@ static int ps3fb_shutdown(struct ps3_system_bus_device *dev)
                 info = dev->core.driver_data = NULL;
         }
         iounmap((u8 __force __iomem *)ps3fb.dinfo);
+        lv1_gpu_fb_close(ps3fb.context_handle);
         lv1_gpu_context_iomap(ps3fb.context_handle, GPU_IOIF, xdr_lpar,
                               ps3fb_videomemory.size, CBE_IOPTE_M);
         lv1_gpu_context_free(ps3fb.context_handle);