Fix tests copy paste