Minor cleanup for inject call