Fixing AttributeError

Hi, I found that your test.py has a problem while I'm testing it.
in test.py, the error is: 
'B' object has no attribute 'n', in line 15.
It must have been class error.
so I fixed this and pulled an request :) # #