Set the mock controller in the mock only if it's not already set.