Fix tests and release task