Make visualize_controller_test pass again
Default weights to the last weight lifted