Update test.py

Fixed failing test