Update more code tests