Fix tests. >_>