Hell yeah fixed that damn bug

Yeah son.  Also added the test.py