diff --git a/codecov.yml b/codecov.yml
deleted file mode 100644
index 7585bee..0000000
--- a/codecov.yml
+++ /dev/null
@@ -1,2 +0,0 @@
-ignore:
-  - internal/colour
diff --git a/internal/colour/colour.go b/internal/colour/colour.go
deleted file mode 100644
index 3379c5e..0000000
--- a/internal/colour/colour.go
+++ /dev/null
@@ -1,27 +0,0 @@
-// Package colour implements basic text colouring for showing text diffs.
-package colour
-
-import (
-	"github.com/fatih/color"
-)
-
-var (
-	header = color.New(color.FgCyan, color.Bold)
-	green  = color.New(color.FgGreen)
-	red    = color.New(color.FgRed)
-)
-
-// Header returns a diff header styled string.
-func Header(text string) string {
-	return header.Sprint(text)
-}
-
-// Green returns a green styled string.
-func Green(text string) string {
-	return green.Sprint(text)
-}
-
-// Red returns a red styled string.
-func Red(text string) string {
-	return red.Sprint(text)
-}
diff --git a/test.go b/test.go
index c8cd56d..f74e651 100644
--- a/test.go
+++ b/test.go
@@ -15,8 +15,14 @@ import (
 	"sync"
 	"testing"
 
-	"github.com/FollowTheProcess/test/internal/colour"
 	"github.com/FollowTheProcess/test/internal/diff"
+	"github.com/fatih/color"
+)
+
+var (
+	header = color.New(color.FgCyan, color.Bold)
+	green  = color.New(color.FgGreen)
+	red    = color.New(color.FgRed)
 )
 
 // Equal fails if got != want.
@@ -327,6 +333,7 @@ func False(tb testing.TB, got bool, options ...Option) {
 func Diff(tb testing.TB, got, want string) {
 	tb.Helper()
 
+	// TODO(@FollowTheProcess): If either got or want don't end in a newline, add one
 	if diff := diff.Diff("want", []byte(want), "got", []byte(got)); diff != nil {
 		tb.Fatalf("\nDiff\n----\n%s\n", prettyDiff(string(diff)))
 	}
@@ -434,19 +441,25 @@ func CaptureOutput(tb testing.TB, fn func() error) (stdout, stderr string) {
 
 // prettyDiff takes a string diff in unified diff format and colourises it for easier viewing.
 func prettyDiff(diff string) string {
+	// color by default will look at whether stdout is a tty and the value of the
+	// $NO_COLOR env var, we need to override this because go test buffers output
+	// so it will appear as if it's not a tty, even though the end result is to show
+	// the output in a terminal. It still respects the value of $NO_COLOR.
+	color.NoColor = false || os.Getenv("NO_COLOR") != ""
+
 	lines := strings.Split(diff, "\n")
 	for i := 0; i < len(lines); i++ {
 		trimmed := strings.TrimSpace(lines[i])
 		if strings.HasPrefix(trimmed, "---") || strings.HasPrefix(trimmed, "- ") {
-			lines[i] = colour.Red(lines[i])
+			lines[i] = red.Sprint(lines[i])
 		}
 
 		if strings.HasPrefix(trimmed, "@@") {
-			lines[i] = colour.Header(lines[i])
+			lines[i] = header.Sprint(lines[i])
 		}
 
 		if strings.HasPrefix(trimmed, "+++") || strings.HasPrefix(trimmed, "+ ") {
-			lines[i] = colour.Green(lines[i])
+			lines[i] = green.Sprint(lines[i])
 		}
 	}
 
diff --git a/test_test.go b/test_test.go
index b7c0c71..d5dcb24 100644
--- a/test_test.go
+++ b/test_test.go
@@ -434,6 +434,7 @@ func TestTest(t *testing.T) {
 		t.Run(tt.name, func(t *testing.T) {
 			buf := &bytes.Buffer{}
 			tb := &TB{out: buf}
+			t.Setenv("NO_COLOR", "true")
 			snap := snapshot.New(t, snapshot.Update(*update))
 
 			if tb.failed {