Added missing return statement... dooooh!