Fix tests errora