[fix] Make sure we process the correct function..