Fix tests for blatant error