Fix small warnings
Update method to match new interface