Put those two tests back in internal_test.go .