Updating tests for new dashboard code