Fix for graph getting the wrong dimensions