diff --git a/internal/bbsim/api/onus_handler.go b/internal/bbsim/api/onus_handler.go
index ce0dba0..1b823d8 100644
--- a/internal/bbsim/api/onus_handler.go
+++ b/internal/bbsim/api/onus_handler.go
@@ -52,7 +52,7 @@
 func (s BBSimServer) GetONU(ctx context.Context, req *bbsim.ONURequest) (*bbsim.ONU, error) {
 	olt := devices.GetOLT()
 
-	onu, err := olt.FindOnu(req.SerialNumber)
+	onu, err := olt.FindOnuBySn(req.SerialNumber)
 
 	if err != nil {
 		res := bbsim.ONU{}
@@ -87,7 +87,7 @@
 
 	olt := devices.GetOLT()
 
-	onu, err := olt.FindOnu(req.SerialNumber)
+	onu, err := olt.FindOnuBySn(req.SerialNumber)
 
 	if err != nil {
 		res.StatusCode = int32(codes.NotFound)
@@ -132,7 +132,7 @@
 
 	olt := devices.GetOLT()
 
-	onu, err := olt.FindOnu(req.SerialNumber)
+	onu, err := olt.FindOnuBySn(req.SerialNumber)
 
 	if err != nil {
 		res.StatusCode = int32(codes.NotFound)