Fix for Graph element removal.

Signed-off-by: Nathan T Freeman <nathan.freeman@redpilldevelopment.com>