forgot return statement. fixed now