Fix tests. Solution work