Wasn't returning value in previous commit. Wishing this was scala!