Move assert cleanup to the main.m file