Fix the up/down method call on Diagram