fix new world and single_ffnn architecture