Fix for blocks falling off graph