Unfortunately yes, this is (currently) expected. The problem is that B is a global variable and currently we can't analyze the code in the comprehension to ensure that it doesn't change the type of B.
Various suggestions has come up about how to fix this, but currently your options is to make B a local variable (put everything in a function) or declare B as const.
