Fix tests to actually test