Small code improvement

Tweaked to match current Akka.