diff --git a/pkg/nvml/types.go b/pkg/nvml/types.go
index d6ed9c4..e4ae40f 100644
--- a/pkg/nvml/types.go
+++ b/pkg/nvml/types.go
@@ -24,57 +24,57 @@ import (
 //
 //go:generate moq -out nvml_mock.go . Interface
 type Interface interface {
-	Init() Return
-	Shutdown() Return
 	DeviceGetCount() (int, Return)
 	DeviceGetHandleByIndex(Index int) (Device, Return)
 	DeviceGetHandleByUUID(UUID string) (Device, Return)
-	SystemGetDriverVersion() (string, Return)
-	SystemGetCudaDriverVersion() (int, Return)
 	ErrorString(r Return) string
+	Init() Return
+	Shutdown() Return
+	SystemGetCudaDriverVersion() (int, Return)
+	SystemGetDriverVersion() (string, Return)
 }
 
 // Device defines the functions implemented by an NVML device
 //
 //go:generate moq -out device_mock.go . Device
 type Device interface {
-	GetIndex() (int, Return)
-	GetPciInfo() (PciInfo, Return)
-	GetMemoryInfo() (Memory, Return)
-	GetUUID() (string, Return)
-	GetMinorNumber() (int, Return)
-	IsMigDeviceHandle() (bool, Return)
-	GetDeviceHandleFromMigDeviceHandle() (Device, Return)
-	SetMigMode(Mode int) (Return, Return)
-	GetMigMode() (int, int, Return)
-	GetGpuInstanceProfileInfo(Profile int) (GpuInstanceProfileInfo, Return)
-	GetGpuInstances(Info *GpuInstanceProfileInfo) ([]GpuInstance, Return)
-	GetMaxMigDeviceCount() (int, Return)
-	GetMigDeviceHandleByIndex(Index int) (Device, Return)
-	GetGpuInstanceId() (int, Return)
+	GetAttributes() (DeviceAttributes, Return)
 	GetComputeInstanceId() (int, Return)
 	GetCudaComputeCapability() (int, int, Return)
-	GetAttributes() (DeviceAttributes, Return)
+	GetDeviceHandleFromMigDeviceHandle() (Device, Return)
+	GetGpuInstanceId() (int, Return)
+	GetGpuInstanceProfileInfo(Profile int) (GpuInstanceProfileInfo, Return)
+	GetGpuInstances(Info *GpuInstanceProfileInfo) ([]GpuInstance, Return)
+	GetIndex() (int, Return)
+	GetMaxMigDeviceCount() (int, Return)
+	GetMemoryInfo() (Memory, Return)
+	GetMigDeviceHandleByIndex(Index int) (Device, Return)
+	GetMigMode() (int, int, Return)
+	GetMinorNumber() (int, Return)
 	GetName() (string, Return)
+	GetPciInfo() (PciInfo, Return)
+	GetUUID() (string, Return)
+	IsMigDeviceHandle() (bool, Return)
+	SetMigMode(Mode int) (Return, Return)
 }
 
 // GpuInstance defines the functions implemented by a GpuInstance
 //
 //go:generate moq -out gi_mock.go . GpuInstance
 type GpuInstance interface {
-	GetInfo() (GpuInstanceInfo, Return)
-	GetComputeInstanceProfileInfo(Profile int, EngProfile int) (ComputeInstanceProfileInfo, Return)
 	CreateComputeInstance(Info *ComputeInstanceProfileInfo) (ComputeInstance, Return)
-	GetComputeInstances(Info *ComputeInstanceProfileInfo) ([]ComputeInstance, Return)
 	Destroy() Return
+	GetComputeInstanceProfileInfo(Profile int, EngProfile int) (ComputeInstanceProfileInfo, Return)
+	GetComputeInstances(Info *ComputeInstanceProfileInfo) ([]ComputeInstance, Return)
+	GetInfo() (GpuInstanceInfo, Return)
 }
 
 // ComputeInstance defines the functions implemented by a ComputeInstance
 //
 //go:generate moq -out ci_mock.go . ComputeInstance
 type ComputeInstance interface {
-	GetInfo() (ComputeInstanceInfo, Return)
 	Destroy() Return
+	GetInfo() (ComputeInstanceInfo, Return)
 }
 
 // GpuInstanceInfo holds info about a GPU Instance