Now properly returns the function when on the declaration