Fixed the bug with not getting back from the dimension.