Adapt function to actual input